Skip to content

Conversation

@Lauenburg
Copy link
Contributor

@Lauenburg Lauenburg commented Jan 29, 2022

I added multiple switches to dataset_volume.py that ensure its functionality in the case that labels and/or valid_mask is set to None.

I tested my adaptation with dummy data for the following cases:

  • mode='train', volume=data, lable=data, valid_mask=data

  • mode='train', volume=data, lable=None, valid_mask=data

  • mode='train', volume=data, lable=data, valid_mask=None

  • mode='train', volume=data, lable=None, valid_mask=None

  • mode='test', volume=data, lable=data, valid_mask=data

  • mode='test', volume=data, lable=None, valid_mask=data

  • mode='test', volume=data, lable=data, valid_mask=None

  • mode='test', volume=data, lable=None, valid_mask=None

  • mode='val', volume=data, lable=data, valid_mask=data

  • mode='val', volume=data, lable=None, valid_mask=data

  • mode='val', volume=data, lable=data, valid_mask=None

  • mode='val', volume=data, lable=None, valid_mask=None

…ata loader in case of labels and/or valid_mask is set to None
@zudi-lin zudi-lin merged commit 5772861 into PytorchConnectomics:master Jan 30, 2022
@zudi-lin zudi-lin linked an issue Jan 30, 2022 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Bug: AttributeError in VolumeDataset when not providing a label list

2 participants